FileHandlingHttpRequestExecuterImpl.java

/*
 ###############################################################################
 #                                                                             #
 #    Copyright (C) 2011-2016 OpenMEAP, Inc.                                   #
 #    Credits to Jonathan Schang & Rob Thacher                                 #
 #                                                                             #
 #    Released under the LGPLv3                                                #
 #                                                                             #
 #    OpenMEAP is free software: you can redistribute it and/or modify         #
 #    it under the terms of the GNU Lesser General Public License as published #
 #    by the Free Software Foundation, either version 3 of the License, or     #
 #    (at your option) any later version.                                      #
 #                                                                             #
 #    OpenMEAP is distributed in the hope that it will be useful,              #
 #    but WITHOUT ANY WARRANTY; without even the implied warranty of           #
 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            #
 #    GNU Lesser General Public License for more details.                      #
 #                                                                             #
 #    You should have received a copy of the GNU Lesser General Public License #
 #    along with OpenMEAP.  If not, see <http://www.gnu.org/licenses/>.        #
 #                                                                             #
 ###############################################################################
 */

package com.openmeap.http;

import java.io.File;
import java.net.FileNameMap;
import java.net.URLConnection;
import java.nio.charset.Charset;
import java.util.Hashtable;
import java.util.Map;

import org.apache.http.HttpVersion;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.mime.HttpMultipartMode;
import org.apache.http.entity.mime.MultipartEntity;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.params.CoreProtocolPNames;

import com.openmeap.constants.FormConstants;

public class FileHandlingHttpRequestExecuterImpl extends HttpRequestExecuterImpl {
	
	@Override
	public HttpResponse postData(String url, Hashtable getParams, Hashtable postParams) throws HttpRequestException {
		
		// test to determine whether this is a file upload or not.
		Boolean isFileUpload = false;
		for( Object o : postParams.values() ) {
			if( o instanceof File ) {
				isFileUpload = true;
				break;
			}
		}
		
		if( isFileUpload ) {
			try {
				HttpPost httpPost = new HttpPost(createUrl(url,getParams));
				
				httpPost.getParams().setParameter(CoreProtocolPNames.PROTOCOL_VERSION, HttpVersion.HTTP_1_1);
				
				MultipartEntity entity = new MultipartEntity(HttpMultipartMode.BROWSER_COMPATIBLE);
		
				for( Object o : postParams.entrySet() ) {
					Map.Entry<String, Object> entry = (Map.Entry<String, Object>)o;
					
					if( entry.getValue() instanceof File ) {
						
						// For File parameters
						File file = (File)entry.getValue();
						FileNameMap fileNameMap = URLConnection.getFileNameMap();
						String type = fileNameMap.getContentTypeFor(file.toURL().toString());
						
						entity.addPart( entry.getKey(), new FileBody((( File ) entry.getValue() ), type ));
					} else {
						
						// For usual String parameters
						entity.addPart( entry.getKey(), new StringBody( entry.getValue().toString(), "text/plain", Charset.forName( FormConstants.CHAR_ENC_DEFAULT )));
					}
				}
					
				httpPost.setEntity(entity);
			
				return execute(httpPost);
			} catch(Exception e) {
				throw new HttpRequestException(e);
			}
		} else {
			
			return super.postData(url,getParams,postParams);
		}
	}
}